package com.lance.answer.question_answer;

import com.alibaba.fastjson.JSONObject;
import com.lance.answer.gui.answer.AnswerObject;
import com.lance.answer.gui.answer.Util;
import com.lance.answer.util.Constant;
import com.lance.answer.util.MD5Utils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.springframework.util.CollectionUtils;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Slf4j
public class QuestionAnswer4 {

    /**
     * 进度表：
     * 6-19 21:00开始测试 人体解剖生理学 行政管理专科
     * @param args
     * @throws Exception
     */
    public static void main(String[] args) throws Exception {
        //学科中文名
      //  String subjectName = "人体解剖生理学";
        String subjectName = "医学免疫学与微生物学";
        //专业中文名
        String majorName = "药学";
        //并发开启的浏览器数量
        int sync = 6;
        //是否提交答案
        boolean commit = true;
        //驱动路径
        String driverPath = null;

        //初始化日志配置
        Constant.initLog();
        //开启任务
        System.out.println("刷题课程：     " + subjectName);
        System.out.println("刷题专业：     " + majorName);
        System.out.println("并发开启浏览器：" + sync);
        System.out.println("是否提交答案：  " + commit);
        start(subjectName, majorName, sync, driverPath, commit);
    }

    public static void start(String subjectName, String majorName, int sync, String driverPath, boolean commit) throws Exception {
        //获取模板路径,通过流的方式获取文件可以避免被打成jar包之后无法获取文件路径的异常
        InputStream inputStream = Thread.currentThread().getContextClassLoader().getResourceAsStream(subjectName + "/" + subjectName + "-answer.json");
        String answerStr = IOUtils.toString(inputStream);

        List<AnswerObject> list = JSONObject.parseArray(answerStr, AnswerObject.class);
        Map<String, AnswerObject> map = new HashMap<>();
        for (AnswerObject answerObject : list) {
            map.put(answerObject.getQuestionCode(), answerObject);
        }

        InputStream inputStream2 = Thread.currentThread().getContextClassLoader().getResourceAsStream(subjectName + "/" + "users/" + majorName + ".user");
        List<String> users = IOUtils.readLines(inputStream2, "utf-8");

        InputStream inputStream3 = Thread.currentThread().getContextClassLoader().getResourceAsStream(subjectName + "/" + subjectName + "-quiz.section");
        List<String> list1 = IOUtils.readLines(inputStream3);
        List<String> paths = new ArrayList<>();
        for (String str : list1) {
            paths.add(str.split(",")[0].trim());
        }

        List<List<String>> group = new ArrayList<>();
        int size = users.size();
        int step = size / sync;
        if (step == 0) {
            group.add(users);
        } else {
            for (int i = 0; i < sync; i++) {
                List<String> list2 = users.subList(step * i, (i + 1) * step);
                group.add(list2);
            }
            group.add(users.subList(step * sync, users.size()));
        }
        for (List<String> batchUser : group) {
            Constant.THREAD_POOL_BATCH.execute(() -> {
                try {
                    for (String user : batchUser) {
                        String[] strs = user.split(",");
                        String userCode = strs[0];
                        String pwd = strs[1].substring(6, 14);
                        run(paths, map, subjectName, driverPath, userCode, pwd, commit);
                    }

                } catch (IOException e) {
                    e.printStackTrace();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });

        }


    }


    // 管理学基础 2065001457555 19940419
    public static void run(List<String> urlTasks, Map<String, AnswerObject> answers, String subjectName, String driverPath, String userName, String password, boolean commit) throws Exception {
        WebDriver driver = Util.getDriver(driverPath);
        //String url = "http://student.ouchn.cn/";
        String url = "http://passport.ouchn.cn/Account/Login?ReturnUrl=%2Fconnect%2Fauthorize%2Fcallback%3Fclient_id%3Dstudentspace%26redirect_uri%3Dhttp%253A%252F%252Fstudent.ouchn.cn%252F%2523%252Fsignin-oidc%2523%26response_type%3Did_token%2520token%26scope%3Dopenid%2520profile%2520ouchnuser%2520ouchnstudentspaceapi%2520offline_access%26state%3Dca1cb5d0270b49fdaea508b8a846c457%26nonce%3Dd257df51b0a141ac8ed35b1caf59ef4c";
        driver.get(url);
        Thread.sleep(3000);
        //Util.login(driver, "2065001457555", "19940419");
        Util.login(driver, userName, password);
        Util.switchToFirst(driver);
        Thread.sleep(3000);
        // 形考作业的链接 管理学
        //List<String> urlTasks = Util.getCourseTaskUrl(driver, CourseTypeEnum.BasicManagement.name);
        //List<String> urlTasks = Util.getCourseTaskUrl(subjectCode);

        //chromeDriver.navigate().refresh();
        for (WebElement we : driver.findElements(By.className("media-body"))) {
            Util.switchToFirst(driver);
            // 找到管理学基础
            String mediaTitle = we.findElement(By.className("media-title")).getText();

            if (!mediaTitle.contains(subjectName)) {
                continue;
            }
            //点击进入学习的按钮
            WebElement studyButton = we.findElement(By.cssSelector("button[class='btn bg-primary']"));
            studyButton.click();
            //Util.switchToLast(driver);
            try {
                Thread.sleep(5000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

        System.out.println(urlTasks.toString());
        for (String urlTask : urlTasks) {
            // FIXME 当完成第一份形考任务时，必须提交，否则会弹出 ”是否保存当前页面“消息
            driver.get(urlTask);
            Thread.sleep(2 * 1000);
            // 点击继续作业

            boolean isGoOn = true;
            for(WebElement we:driver.findElements(By.cssSelector("*[class='btn btn-secondary']"))){
                if("重新试答测验".equals(we.getText())){
                    isGoOn = false;
                }
            }
            for(WebElement we:driver.findElements(By.tagName("a"))){
                if("回顾".equals(we.getText())){
                    isGoOn = false;
                }
            }
            if(!isGoOn){
                continue;
            }

            try {
                WebElement doWorkArea = driver.findElement(By.cssSelector("div[class='box py-3 quizattempt']"));
                doWorkArea.findElement(By.cssSelector("button[class='btn btn-secondary']")).click();
            } catch (Exception e) {
                System.err.println("taskUrl:" + urlTask + "  occur error");
                e.printStackTrace();
            }
            Thread.sleep(2 * 1000);
            //回到试题首页
            WebElement element = driver.findElement(By.id("quiznavbutton1"));
            element.click();
            Thread.sleep(2 * 1000);
            // 回到首页
            //Util.goHomePageIndex(driver);
            boolean hashNextPage = true;
            while (hashNextPage) {
                try {
                    Thread.sleep(4000);
                    // 单个页面的所有题目的集合
                    List<WebElement> singlePageQuestion = Util.getSinglePageQuestionList(driver);
                    // 遍历每个问题|题目


                    for (WebElement singleSubject : singlePageQuestion) {

                        WebElement contentElement = singleSubject.findElement(By.className("content"));

                        String title ;
                        String answerClass = singleSubject.getAttribute("class");
                        if (answerClass.contains("multianswer")) {
                            title = Util.getTitle(contentElement);
                        }else {
                            // qtext 题目标题
//                            WebElement qTitleElement = contentElement.findElement(By.className("qtext"));
//                            // 题目名称
//                            title = qTitleElement.getText();
                            title = Util.getTitle(contentElement);
                        }
                        //兼容特殊字符
                        title = Util.dealContent(title);

                        String questionCode = MD5Utils.encode(subjectName + title);

                        if (!answers.containsKey(questionCode)) {
                            System.out.println("答案不存在");
                            System.out.println(title);
                            continue;
                        }

                        AnswerObject answerObject = answers.get(questionCode);

                        WebElement answerElement = Util.findByClassName(contentElement, "answer");
                        if (null != answerElement) {
                            // 开始答题
                            Util.doExercise(driver, answerElement, answerObject.getQuestionType(), answerObject.getAnswers());
                        }else {
                            // 如果为空就是填空题 是没有答案对比的
                            Util.doExercise(driver, singleSubject, answerObject.getQuestionType(), answerObject.getAnswers());
                        }
                        //  FIXME 此时答案是没有保存的，要点击结束试答题
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    Thread.sleep(2000);
                    // FIX ME 获取当前页面的下一页信息 设置为while还是有下一页就继续循环
                    WebElement nextPage = Util.findNextPage(driver);
                    if (nextPage != null) {
                        nextPage.click();
                    } else {
                        hashNextPage = false;
                        // FIXME  此时答案是没有保存的，要点击结束试答题 当没有下一页的时候，可以选择提交
                    }
                }
            }


            Thread.sleep(2000);

            //回到试题首页 (保存最后一题答案)
           /* WebElement element2 = driver.findElement(By.id("quiznavbutton1"));
            element2.click();*/

            WebElement submitbtns = driver.findElement(By.className("submitbtns"));
            WebElement next = submitbtns.findElement(By.cssSelector("input[name='next']"));
            next.click();
            Thread.sleep(2000);

            if (!commit) {
                log.info("答案不提交");
                TimeUnit.SECONDS.sleep(2);
                continue;
            }

            //提交答案
            List<WebElement> saveButtons = driver.findElements(By.cssSelector("button[type='submit']"));
            //System.out.println("提交");
            //System.out.println(saveButtons.size());
            if (!CollectionUtils.isEmpty(saveButtons)) {
                for (WebElement webElement : saveButtons) {
                    String text = webElement.getText();
                    if (text.contains("提交")) {
                        webElement.click();
                    }
                }
            }

            TimeUnit.SECONDS.sleep(3);
            //全部提交
            //提交答案

            List<WebElement> confirmation = driver.findElements(By.className("confirmation-dialogue"));
            //System.out.println("确认弹框");
            //System.out.println(confirmation.size());
            //System.out.println(confirmation.toString());
            if (!CollectionUtils.isEmpty(confirmation)) {
                for (WebElement webElement : confirmation) {
                    List<WebElement> submitElement = webElement.findElements(By.tagName("input"));
//                    System.out.println("再次确认按钮");
//                    System.out.println(submitElement.size());
                    if (!CollectionUtils.isEmpty(submitElement)) {
                        for (WebElement element1 : submitElement) {
                            String value = element1.getAttribute("value");
//                            System.out.println("不是空的");
//                            System.out.println(value);
//                            System.out.println("按钮长这样");
//                            System.out.println(element1.toString());
                            if (value.contains("提交")) {
                                //点击确认后页面会刷新,此时为了防止丢到元素抛异常，这里直接获取第一个，不再循环
                                element1.click();
                                break;
                            }

                        }
                    }
                }
            }


            List<WebElement> submitButtons = driver.findElements(By.tagName("input"));
            //System.out.println("空的吗");
            //System.out.println(submitButtons.toString());
            if (!CollectionUtils.isEmpty(submitButtons)) {
                for (WebElement webElement : submitButtons) {
                    String text = webElement.getText();
                    //System.out.println("不是空的");
                    //System.out.println(text);
                    if (text.contains("提交")) {
                        webElement.click();
                    }
                }
            }

            TimeUnit.SECONDS.sleep(2);
        }

        Thread.sleep(5000);
        driver.close();
        driver.quit();
    }


}
